package com.liy.redisson.utils;

import cn.hutool.http.HttpUtil;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;

import javax.servlet.http.HttpServletRequest;

/**
 * @author ly
 * @date 2022/11/11 17:56
 * @Description 根据ip获取归属地
 * @Version 1.0
 */
@Slf4j
public class Ip {

    /**
     * 获取IP地址
     *
     * 使用Nginx等反向代理软件， 则不能通过request.getRemoteAddr()获取IP地址
     * 如果使用了多级反向代理的话，X-Forwarded-For的值并不止一个，而是一串IP地址，X-Forwarded-For中第一个非unknown的有效IP字符串，则为真实IP地址
     */
    public static String getIpAddr(HttpServletRequest request) {
        String ip = null;
        try {
            ip = request.getHeader("x-forwarded-for");
            if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) {
                ip = request.getHeader("Proxy-Client-IP");
            }
            if (StringUtils.isEmpty(ip) || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
                ip = request.getHeader("WL-Proxy-Client-IP");
            }
            if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) {
                ip = request.getHeader("HTTP_CLIENT_IP");
            }
            if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) {
                ip = request.getHeader("HTTP_X_FORWARDED_FOR");
            }
            if (StringUtils.isEmpty(ip) || "unknown".equalsIgnoreCase(ip)) {
                ip = request.getRemoteAddr();
            }
        } catch (Exception e) {
            log.error("IPUtils ERROR ", e);
        }

        //使用代理，则获取第一个IP地址
        if(StringUtils.isEmpty(ip) ) {
            if(ip.indexOf(",") > 0) {
                ip = ip.substring(0, ip.indexOf(","));
            }
        }

        return ip;
    }

    /**
     * 根据ip获取归属地
     * @param ip
     * @return
     */
    public static IpAddress getAddress(String ip) {
        String url = "http://freeapi.ipip.net/" + ip;
        String str = HttpUtil.get(url);
        log.info(str);
//        if(!StrUtil.hasBlank(str)){
//            String substring = str.substring(str.indexOf("{"), str.indexOf("}")+1);
//            JSONObject jsonObject = JSON.parseObject(substring);
//            String province = jsonObject.getString("province");
//            String city = jsonObject.getString("city");
//            IpAddress ipAddress = new IpAddress();
//            ipAddress.setProvince(province);
//            ipAddress.setCity(city);
//            System.out.println("ip："+ ip + "，省份：" + province + "，城市：" + city);
//            return ipAddress;
//        }
        return null;
    }

    @Data
    public static class IpAddress{
        private String province;
        private String city;
    }

    public static void main(String[] args) {
        System.out.println(getAddress("127.0.0.1"));
    }

}
